package cc.wsyw126.java.nowcoder.interview;

import java.util.ArrayList;
import java.util.LinkedList;

/**
 * Created by Administrator on 2017/4/26.
 * 有家动物收容所只收留猫和狗，但有特殊的收养规则，收养人有两种收养方式，第一种为直接收养所有动物中最早进入收容所的，第二种为选择收养的动物类型（猫或狗），并收养该种动物中最早进入收容所的。
 * 给定一个操作序列int[][2] ope(C++中为vector<vector<int>>)代表所有事件。
 * 若第一个元素为1，则代表有动物进入收容所，第二个元素为动物的编号，正数代表狗，负数代表猫；
 * 若第一个元素为2，则代表有人收养动物，第二个元素若为0，则采取第一种收养方式，若为1，则指定收养狗，若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作，即没有可以符合领养要求的动物，则将这次领养操作忽略。
 * 测试样例：
 * [[1,1],[1,-1],[2,0],[2,-1]]
 * 返回：[1,-1]
 */
public class CatDogAsylum {
    public ArrayList<Integer> asylum(int[][] ope) {
        ArrayList<Integer> result = new ArrayList<>();
        LinkedList<node> cat = new LinkedList<>();
        LinkedList<node> dog = new LinkedList<>();

        int position = 0;
        for (int i = 0; i < ope.length; i++) {
            if (ope[i][0] == 1) {
                if (ope[i][1] > 0) {
                    dog.add(new node(ope[i][1], position++));
                } else {
                    cat.add(new node(ope[i][1], position++));
                }
            } else if (ope[i][0] == 2) {
                node catCheck = null;
                node dogCheck = null;
                if (!dog.isEmpty()) dogCheck = dog.getFirst();
                if (!cat.isEmpty()) catCheck = cat.getFirst();
                if (dogCheck == null && catCheck == null) continue;
                if (ope[i][1] == 0) {
                    if (dogCheck != null && catCheck != null) {
                        if (dogCheck.position<catCheck.position) {
                            result.add(dogCheck.x);
                            dog.poll();
                        } else {
                            result.add(catCheck.x);
                            cat.poll();
                        }
                    } else if (dogCheck == null ){
                        result.add(catCheck.x);
                        cat.poll();
                    } else if (catCheck == null) {
                        result.add(dogCheck.x);
                        dog.poll();
                    }
                }else if (ope[i][1] == 1) {
                    if (dogCheck != null) {
                        result.add(dogCheck.x);
                        dog.poll();
                    }
                } else if (ope[i][1] ==-1) {
                    if (catCheck!=null) {
                        result.add(catCheck.x);
                        cat.poll();
                    }
                }
            }
        }
        return result;
    }

    private class node {
        int x;
        int position;

        public node(int x, int position) {
            this.x = x;
            this.position = position;
        }
    }

    public static void main(String[] args) {
        CatDogAsylum catDogAsylum = new CatDogAsylum();
        ArrayList<Integer> asylum = catDogAsylum.asylum(new int[][]{{1, 1}, {1, -1}, {2, 0}, {2, -1}});
        asylum.forEach(System.out::print);
    }
}
